Worked on assignment 5