Made some changes to setup.py