Added posterior data randomization when pruning classes. SGD was performing very poorly