Lesson 2 - added categories