Update mainClass.java

I have 0 clue if this compiles. Hope for the best.