update the python example