Updated docstrings and more.