Python 2 & 3 compatibility

Python 2 & 3 compatibility