edited plot python sript