Add plot.py