Update to setup.py