调整代码

最后调整代码。现在可以生成3个文件。需要后续完善