Update README.md

adding description specifying python code