updated plot.py