TDD small exercise