[TASK] Update badge styles