Update 2 for Lesson 2