small change to MainClass