HW4 update