Add main class