Fix python example so it runs
Merge pull request #290 from tubaman/fix_python_example

Fix python example so it runs