Update setup.py for Python3 compatibility