:gem: Change Chat UI code