Cleaning up lesson 1 homework