Add the highlighted nodes shown in the images below, locations are at the top of the screenshots.