Update README.md
Try to write a python implementation