Making changes on protocol.py