Update README
Add some docstrings