Edit main.java