Small changes to setup.py