UI fixes

ui