Updates to the setup.py