Small change to MainClass