Add possibility for custom notification handling