Just update main.java