Updated Main.java