Adjust menu, add actual code