browser.runtime.onMessage.addListener((message, sender) => {
    if (message.action === 'updateBadge' && sender.tab) {
        const count = message.count > 0 ? message.count.toString() : '';
        
        browser.action.setBadgeText({
            text: count,
            tabId: sender.tab.id
        });

        browser.action.setBadgeBackgroundColor({
            color: '#486860',
            tabId: sender.tab.id
        });
    }
});