Modified Main.class