improve benchmark mode