Basic support for both python2 and python3