Fix main class selection menu in run