Update example.py