I think this should be in the Notification model