Update template to use notification