TDD Exersice for class