Add comments in Main.java