Worked on chat UI