Update communication.py