I am just putting the changes done for Assignment2