Updated Message class