Added methods to TestClass