softmax function fix