worked on HW 4