Updated imports in docstrings