docstring fixes and import fix