Modified util.py