Improved docstring and README.md