Using correct way to do predict
Working in progress, tempararily save