Minor edits to Main.java