add more code

Main.java file