Make softmax not overflow