Modify setup.py