Additions to plots

Added some code to create particular plots. Still a bit messy.