Update util.py

format code