修复bug，完善功能