Add prediction file saving