Lesson 5 solution fixed