Improve README
Docstrings