HW4 Step 2