changed loss to the correct one in the softmax vector