Adding plot 4 and commenting/cleaning up