Add benchmark code