Add valid softmax