Fix main.class