write the softmax