add docstring
fix whitespace