Improve chat example