add badge and warning