Adding notification feature