Update code for message