Updated TestClass