代码优化

Signed-off-by: Zealot Ke <chzealot@gmail.com>