Modify the main.java