add more code

modify test.java file