TDD first attempt