Small TDD exercise