Added TDD tests using should