Add setup.py
Update __init__.py