Removed unused imports from main class