made a few tweaks, fixed the last plot as I was doing it a bit wrong