Added set message method