changes to docstrings and imports