Fix setup.py for PyPI