Fixed error in softmax