update bench.py