A few modifications to setup.py