Remove unused imports from main class