Fix README python example