Updated docstrings and README