Add callback for session end