adds softmax function