修复bug，完善代码