Added a test to check that the softmax cost is unaffected by the change