additional tests and a slight improvement of softmax