HW6 Part 4 - Step 2