Update setup.py to new version.py api