Update util.py
Merge pull request #9 from chengguopiao/patch-10

Update util.py