Fix name of main class