bugs in softmax, printing