Updated for new release to PyPi