Update __init__.py
Update setup.py