Cleaned up Docstrings inside tests