Modularizing menus into the unit class