Minor changes to setup.py