Add the main class