Update the menu code to allow for different menu layouts.