Updated docstring and README