Add notification function