You can edit the UIFlow2 python code to add "retain=True" as that functionality is available in the python library. What is needed is that retain= needs to be exposed in the Block definitions. Otherwise, I'm constantly having to add that back in every time I turn off editing mode.