clean up main class