Updated setup for pypi