Add methods to get message type