Updates to reflect publication to pypi