Small updates to benchmark script.