docstring fixes
pep8 inprovements