Update example.py for Python 3 (refs #3)
Fix import (closes #3)

Thanks dzen.