Fixing setup.py for Python 3