add in a prediction (not sparating the data into training and test yet)