整理了代码，简化程序，做了几个function

对function的功能进行测试，并简化了程序。