step 2 edit code