Addition of the test class