Fixed softmax grad and added test