First draft of the predictor. Currently can train a data set, test on a data set and print the percentage of correct predictions