优化了下结构，改了个小bug