Changing setup for python2/python3 compatibility