Update README.md

In this update I've actually added the Python code.