Worked on week 3-5 HW