Update setup.py for new pypi package.