softmax fixes