Remove code for benchmark