Made changes to setup.py