Update control_util.py