HW4 step 2