Update example.py

Bug Fixed