extracted logic to class per traffic source