change to python 2, can support python 3 at same time