Cleanup main class