Improving setup.py to support PyPI publication