Add code for submenus