* do initial run on full context * decrement while loop iterations * add context to output * remove first param * removing first param: change shape invariant