Added the prediction toolbox