Minor documentation change + removal of unused imports.