Update README.md and add python script