Updated example.py to be in line  with example from README.md