Remove 'change in activations' method; can use 'sorted_avg_diff'.
Allow plot_activations to take in an OrderedDict.