Update util.py
Merge pull request #13 from GuanjunXu/patch-13

Update util.py