added some TDD