HW4 no 2 stage 2