Assignment 2

Updated my code and added comments