Modified the docstring in the __init__.py file