Lesson 5 improvements
Lesson 6