add softmax and its unit test