HW4 step 4