Separated the notification code from the base viewmodel