Updating the setup for pypi