update docstring
pep8