Added plot.py