adding notification functionality