Updated methods for UpdateMessage.java