Added Test to the test class