Added one more class to do TDD on