Add softmax unit