update main.py

do some updates in these file
增加了一下输出信息，以便查看在哪里花费了很多时间