Set up main class