I'm working on more UNOPTFLAT fixes, but those require building a control flow graph for always blocks. AstWhile and AstJumpBlock are a bit of a pain. Can we simplify them along the following lines?