Update setup.py for pypi