Update the setup.py