Simplified softmax_pred.m