Add setup
Update setup.py