Edit to setup.py