Updates to setup.py