Add notification type to model