Update main.java

// Good job