Worked on HW5 question