Slight edit of setup.py