Add proper badge