My code for Course 1 Lesson 2