[update] Optimize code