main.java fixes