UI fixes

UI fixes