Script now outputs a prediction file