Update message_state.py