doctest and docstring updates