HW2, completed step 5