Made changes for Assignment 2

Commiting for review