Add little logic
添加了一点点逻辑