Updated docstrings and docs