Add benchmarking code back in for demonstrative purposes