getting softmax to work