HW4 step 1-2