Update util.py