so far at step 4 from the homework