UI fixes + chat fix