Update example.py

Fixing #8