Fixing python example to go with 1.0