fixed up the main class