HW6 - Completed step 1