Add benchmarking code