改进函数， 全部统一到main