Solution to assignment. Still need to add comments.
Added first draft of comments
Added comments at top of file