Remove unused imports
Updated documentation