Step 2 for exercise 3.1.5