Fixed setup.py to support Python 3