Assignment 2 with comments

Added comments to the code for final assignment