Address review comments: Various docstring fixes.