Plotting code added to repo