UI bugs fixed