Fixing some docstrings