Update to speed up the predictions slightly