Moved in and cleaned up some workflow ui controller logic into related nodes behavior where it belongs