Added notification code