Add softmax and phone diagram