code review

[duanzhiihao]

Your code is great. But every time I run your code, I must run 4 .py files seperately. It's not convenient.
Besides, I want to import your module and use your functions but you didn't packege your codes into functions. I think you'd better do so next time.

[feiyue33]

Thanks for your advice! I improved some details according to your suggestions.