little code tweak using .count()