update some code

update some code