Kata 8: Refactor tests