Code for assignment 3 is done