Fixed and added tests for computing a desired return type (value, probability) in Model.