Add boolean in main class