add comment and docstring describing function return