Python 3 compatibility in setup