增加两个测试用例
进一步简化代码，修复bug