removed unused imports
added example in docstring