HW4 Part 2 changes