add docstring to function