Update setup.py
Add init