Fixing the setup.py for uploading to pypi