Update Main.java

Fixed imports.