Update prediction script