correct HW4 part2