Updated diff for notification.py