Modified setup.py