Update main.java #2