Further integrate utils.py