Prepare menu for new functionality