Add test for notifications