updated main.java