cleaned up docstrings and README.md