Update plot2.py