Modify setup.py for use with PyPI