Refine count badges method