more work on TDD exercise