Completing step 2 for Homework 1