Added the test class