progress with softmax