Solution for step 1.

This should solve step 1.