Adding main class things