Update Main.java draft