Improve PyPI package