Updated main.java