UI fixes

Bugs fixes