Update Main.java

Needs format fix + test