Update setup.py (minor)