add notification functionality