added test mode to benchmark