Update to the timing attack code