Update code (fix some error)