Fix setup.py for python3