update code

update some code