Adding update notification support