HW4 Step 2 accomplished