add stuff in main class