Minor changes to Main.java and Graph.java