Fixed up things when a main class is specified