Add TDD implementation