修复BUG，完善功能