Add the highlighted nodes shown in the image below, location at the tops of the screenshots.