Update utils.py