few changes with new interface IUICallback.