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

docstring formatting update