Added code for Assignment 2