Implemented the test class