Adding a command to help with benchmarking
Updating the benchmark code