Some changes to display message, still doesnt work