Fixed HW4 Part 1