Lesson 6: Fix some mistakes