完善功能，修复bug