complete code for Assignment 2