Fixed Plotting code. It should work now!