updated Main.java