editing plot.py