Plotting code

Just playing around with the plotting code