creating python graph