Lesson 5 initial solution