HW4 Step2 accomplished