Adds code for Step 4