update plot.py