Try to write a python implementation
Update README.md