Added comments

Finished assignment with 5 minutes to spare, so added comments afterwards