work on the main class