Main.java updated