[Lesson 4] Refactorying test a bit