Updated HW4 with Wu's corrections