Update example.py

Fixing a wrong comment