修改main

用循环替换了之前的代码。