Allow setting callback for notifications