Clean up chat ui