update the README and the example python code