Update plot.py