Minor updates to benchmarking