more bug fixes with the message display