Update README.md

python code example in README