change to use softmaxact