Added benchmarking code