update TDD exercise