Update TestClass.java

made changes