Lesson 2 Task 2 fix