Began working on the predictor aspect of the model, currently has a bug