first commit of Assign 2 solution