Update example.py and fix __init__.py bugs