removed header for class TrafficGenerator