Removed holdout and created test and train data to predict the values