polish code
update