Updated docstrings for test file