Updated docstrings
Minor update