Update docstrings to reflect improvements made in previous commit