docstring and documentation updates