Update example.py

fixing import