Add comment, small fixes to docstrings.