Edit: menu-badge-integration.php