add more TDD tests