优化功能，修复部分bug