Updated util.py