softmax pseudocode added