Cleaned up chat UI