Add styling for chat unit