Add function in message entity