Fix setup.py for Python 3