Lesson 3 code added