Add benchmark function