[Lesson 4] Add unit test