Adding a benchmark wrapper around the main functionality. In the end, the user should get access to the benchmarks via the model.benchmark object.