Edited the setup.py file