Fixed incorrect docstring and removed unused imports