Added model saving, updated purity graph, added featurization