Cleaning up the docstrings and README