docstring formatting update
missing import
fix broken docstring
Merge pull request #293 from jrjohansson/release-3.1-code-cleanup

docstring formatting update