Assignment 2

Updated Code