Add the benchmarking code
Add the benchmarking code