Updated to use .main-menu