Modify the python script

Delete unnecessary code