added prediction fct