Added notification registration