wrote prediction script