Support python2 and python3