Update for PyPi release