Improve UI behavior.