Adding Softmax to factory