Modified MainClass.java