* update docstrings and tidy code * add docs * address comments * Update __init__.py * address comments
Attach a file by drag & drop or click to upload
