Smart AI – Fixed some errors due to AI being dead

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