Correct imports and docstring in __init__.py