Updated the model for messages