modify Main.java