Fixed the python example