updates to benchmark.py