Update example.py

keep it simple for end user