Add benchmark to test