Add wrapper for UI notification