HW4 updates