Updated Lesson 2 work