added softmax policy