fixed error in softmax