Add setup.py
Update setup.py