Update notification.py