Log softmax bug corrected