Lesson 3 problems fix