put prediction stuff into main