Forgot to add notification model