Benchmarking code added