wrote a softmax classifier