UI fixes
bug fix