Update TestClass.java

little changes