minor changes to Main.java