Fixing up UI.