Entered all the code for assignment 2