Added the plot.py file