I can edit the helper-main-menu.php file.

Can you be more specific about exactly where to place the code you shared above? I tried adding it to the .php file with my opt-in code added, but doing so broke the site (it came up as a blank page).

Thanks.