Trying to do the loop invariant in smt