Added method to the test class