Assignment2

Updated code for Assignment2