add softmax with documentation