Reworked chat UI