Updating main.class