Add in initial training set, testing set, model, dictionary