Updated docstrings from feedback