fixing up UI.