Added softmax test cases