修复一些小bug和优化