fix typo/bug, add docstring