Updated the setup.py