Add simple wrapper for benchmarking