fixed main.java