Update Benchmark/util.py