Add audio to basic chat and improve chat impl