add docstring, allow pep8 style guide