modify main.java