[Brain & Tunji] Adding the UI for the flow with nodes