Simple TDD exercise