Adding a notification system