updated max function to be zero, this makes for a sane softmax when close to zero