Minor updates to the prediction scripts