Update docstrings to reflect documentation changes