Cleaned up docstrings