Code for Assignment