cleans up main class file