Add performance test code