fixing main class