Python 3 compatibility for setup.py