Modify update to include message