Update Main.java

Small tweaks and changes