Update setup.py

Fixed Python 3 issues.