HW6 - Completed step 4