add first code using TDD