Try to write a python implementation
Fix README.md