Tidy some docstrings and comments