Update setup.py to support Python 3