Minor update to setup.py