Adding python 3 compatibility