update code, bug fix.