Update run_benchmarks.py