Spec changed again. Updated end2end test.
Corrected mistake in last commit.
Extracted probability calculation from ArgmaxAssertionModel.