Add missing docstring
Improve README