Add badge class