Added the plotting function, plot1.py