Assignment 2

Updated to add some edits to the comments