Better main class