Update Main.java

Closes #2