add softmax