Update protocol.py