update python codes