updated for pypi release