Allow more free lambdas to be used in the EDSL

Currently forall(a, i -> { return a[i] > 0; }); will cause a runtime error.