HW4 2nd part