Update docstrings as per review.