Pruning works. Need to reorder pruning and normalization