I use softmax instead, thanks! :)