Improved UX when chatting