add plot.py