Modify python2 to python