added prediction model