Add more explanation comments to the chat demo