Inserted my code for assignment 2