Update algo_util.py