kind of working softmax
more softmax stuff