Updated plot.py to be able to plot multiple data sets