Update some docstrings and comments