Fix more docstrings