More updates to comments and docstrings.