Move to python 2