HW4 Update Part 1