HW4 part 4 update