TDD - all test working