Updated code so benchmarks actually run.