Completed assignment

Contains four code files and four plots, as per assignment instructions.
Hopefully well-documented and easy to understand! :)