TDD implemented first 2 tests