改进逻辑和修正功能