updated with our predictions, still have to fix nan values