First implementation of softmax mutliclasse