modify test.java for testing