change the return message method