Update consumer.py