Modified the Main.java