简化代码，完善部分功能