add the functinality on the message showing