Add notification feature