Update docstrings and comments