Modified setup.py file