added functionality to the benchmark method