modify util.py