Step 3 of assignment 1