Made setup.py work with Python2 and Python3