Adding support for Kafka consumer