Add Badge class