Removed unused imports and fixed comment