Fixed imports and docstring snippet