Adjusted benchmark to API.
Merge pull request #115 from simongog/adjust-dr-benchmark

Adjusted benchmark to API.