Slight edit to the Main.class file.