Update setup.py for py3