Add proper badge code