Moved out some workflow ui controller logic into related nodes behavior where it belongs