Removed unused imports
Small code review