Add notification support