update util.py