添加main测试方法，调整原有代码结构