better comment count logic