Correcting main.java