Removed unused imports
Added comments