Small Changes to the Main.java file