Update setup.py

Update setup.py