modify kafka_consumer.py