forget seperate prediction model. added