Add notification features