Fixes #22: final commit to cover randomization tests for softmax