update code python

update code python