Cleaned up the docstrings a bit