Updates to setup.py for the new version