简化代码，更新注释