this should prevent softmax exceeding / deceeding