Minor code formatting fix
Tweaked to use add_node instead of add_menu (add_menu is deprecated)