Create notification.py